#pragma once

#include <iostream>
#include <unistd.h>
#include <sys/wait.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>

#include "../comm/util.hpp"
#include "../comm/log.hpp"

namespace ns_compiler
{
    using namespace ns_log;
    using namespace ns_util;

    class Compiler
    {
    public:
        static bool Compile(const std::string &file_name)
        {
            pid_t id = fork();
            if (id == 0) // 子进程需要执行file_name 同时将错误写入指定文件
            {
                umask(0);

                // 打开 xxx.compile_error文件
                int _stderr = open(PathUtil::CompilerError(file_name).c_str(), O_CREAT | O_WRONLY, 0644);
                if (_stderr < 0)
                {
                    Log(WARNING) << "没有生成err文件"
                                 << "\n";
                    exit(1);
                }

                dup2(_stderr, 2); // 将标准错误的信息出入到_stderr中

                // 程序替换 执行g++
                execlp("g++", "g++", "-o", PathUtil::Exe(file_name).c_str(),
                       PathUtil::Src(file_name).c_str(),"-D","COMPILER_ONLINE","-std=c++11", nullptr); // 最后以nullptr结尾
                Log(ERROR) << "启动g++失败,可能是参数错误"
                           << "\n";
                exit(2);
            }
            else if (id > 0)
            {
                // 父进程需要检查是否生成了可执行程序
                waitpid(id, nullptr, 0);
                if (FileUtil::IsFileExist(PathUtil::Exe(file_name)))
                {
                    Log(INFO) << PathUtil::Src(file_name) << "编译成功"
                              << "\n";
                    return true;
                }
                Log(ERROR) << "编译失败"
                           << "\n";
                return false;
            }
            else
            {
                Log(ERROR) << "内部错误，创建子进程失败"
                           << "\n";
                return false;
            }
        }
    };
}